Skip to content

Enable generating an entry for torch._grouped_mm` for op-level test - #457

Open
kiszk wants to merge 2 commits into
mainfrom
feat/support-grouped_mm
Open

Enable generating an entry for torch._grouped_mm` for op-level test#457
kiszk wants to merge 2 commits into
mainfrom
feat/support-grouped_mm

Conversation

@kiszk

@kiszk kiszk commented Sep 1, 2026

Copy link
Copy Markdown
Collaborator

This PR enables generating an entry for torch._grouped_mm in a YAML file for op-level test. This is required to support Gemma4 26B-A4B

  • skip ops starting with torch.cuda
  • add init: cumsum_offsets
  • skip scalar in meta_val
  • stop deep copies to avoid recursion

Signed-off-by: Kazuaki Ishizaki <ishizaki@jp.ibm.com>
@kiszk
kiszk requested a review from anubhavjana as a code owner September 1, 2026 13:48
@kiszk
kiszk enabled auto-merge September 3, 2026 05:58
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants